import pandas as pd
import os

# 读取CSV文件
file_path = r'D:\2024华为杯数学建模\2024C题-数据驱动与电磁损耗\C题\附件一数据拆解\Q2\Q2磁芯材料4正弦波形.csv'
df = pd.read_csv(file_path, encoding='utf-8', skiprows=1)  # 从第二行开始读取
df.columns = ['温度，oC', '频率，Hz', '磁芯损耗，w/m3', '励磁波形', '磁通密度峰值']  # 设置列名

# 获取第一列的唯一值
unique_values = df['温度，oC'].unique()

# 创建保存文件的目录
output_dir = r'D:\2024华为杯数学建模\2024C题-数据驱动与电磁损耗\C题\附件一数据拆解\Q2\分类结果'
os.makedirs(output_dir, exist_ok=True)

# 遍历每个唯一值，并保存相应的数据
for value in unique_values:
    # 提取相同温度的数据
    subset_df = df[df['温度，oC'] == value]

    # 保存到新文件
    output_file_path = os.path.join(output_dir, f'{value}.csv')
    subset_df.to_csv(output_file_path, index=False, encoding='utf-8')

    print(f"已保存文件: {output_file_path}")

print("所有文件已保存完成。")